import random

import pygame

# 定义屏幕尺寸常量
SCREEN_RECT = pygame.Rect(0, 0, 480, 700)
# 刷新的帧率
FRAME_PER_SEC = 30
# 创建敌机的定时器常量
CREATE_ENEMY_EVENT = pygame.USEREVENT
# 英雄发射子弹定时器常量
HERO_FIRE_EVENT = pygame.USEREVENT + 1


class GameSprite(pygame.sprite.Sprite):
    """飞机大战游戏精灵"""

    def __init__(self, image_name, speed=1):
        # 调用父类初始化方法
        super().__init__()
        # 定义对象属性
        self.image = pygame.image.load(image_name)
        self.rect = self.image.get_rect()
        self.speed = speed

    def update(self):
        # 在屏幕竖直方向上移动
        self.rect.y += self.speed


class BackGround(GameSprite):

    # 创建初始化方法
    def __init__(self, is_alt=False):
        super().__init__('./images/background.png')
        if is_alt:
            self.rect.y = -self.rect.height

    """ 游戏背景精灵 """

    def update(self):
        # 1、调用父类的方法实现
        super().update()

        # 2、判断是否移出屏幕，若移出屏幕，将图像设置到屏幕上方
        if self.rect.y >= SCREEN_RECT.height:
            self.rect.y = -self.rect.height


class Enemy(GameSprite):
    """敌机精灵"""

    def __init__(self):
        # 1、调用父类方法，创建敌机精灵
        super().__init__("./images/enemy1.png")
        # 2、指定敌机的初始随机速度
        self.speed = random.randint(1, 4)
        # 3、指定敌机的初始随机位置
        self.rect.bottom = 0
        max_x = SCREEN_RECT.width - self.rect.width
        self.rect.x = random.randint(0, max_x)

    def update(self):
        # 1、调用父类方法，保持垂直方向飞行
        super().update()
        # 2、判断是否废除屏幕，如果是，需要从敌机精灵组删除
        if self.rect.y >= SCREEN_RECT.height:
            # print("飞机飞出屏幕，从敌机精灵组删除")
            self.kill()

    def __del__(self):
        pass
        # print("敌机销毁。。。%s" % self.rect)


class Hero(GameSprite):
    """英雄精灵"""

    def __init__(self):
        # 1、调用父类方法，设置 image & speed
        super().__init__("./images/me1.png", 5)
        # 2、设置英雄的初始化位置
        self.rect.center = SCREEN_RECT.center
        self.rect.bottom = SCREEN_RECT.bottom - 120
        # 3、创建子弹的精灵组
        self.bullet_group = pygame.sprite.Group()

    def update(self):
        # 英雄在水平位置移动
        self.rect.x += self.speed
        # 控制英雄不能飞出屏幕
        if self.rect.x <= 0:
            self.rect.x = 0
        elif self.rect.right >= SCREEN_RECT.right:
            self.rect.right = SCREEN_RECT.right
        # 控制英雄不能竖直飞出屏幕
        if self.rect.y <= 0:
            self.rect.y = 0
        elif self.rect.bottom >= SCREEN_RECT.bottom:
            self.rect.bottom = SCREEN_RECT.bottom

    def fire(self):
        # 一次性发射三枚子弹
        for i in range(1):
            # 1、创建子弹精灵
            bullet = Bullet()
            # 2、设置精灵位置
            bullet.rect.bottom = self.rect.y - i*20
            bullet.change_color()
            bullet.rect.centerx = self.rect.centerx
            # 3、经精灵添加到精灵组
            self.bullet_group.add(bullet)


class Bullet(GameSprite):
    # 定义子弹精灵
    def __init__(self):
        # 1、调用父类方法，设置子弹图片&初始速度
        super().__init__("./images/bullet1.png", -20)

    def update(self):
        # 1、调用父类方法，让子弹竖直方向飞行
        super().update()

        # 2、判断子弹是否飞出屏幕
        if self.rect.bottom <= 0:
            self.kill()
